package cn.yaboya.system.service.impl;


import cn.yaboya.basic.query.BaseQuery;
import cn.yaboya.basic.service.Impl.BaseServiceImpl;
import cn.yaboya.basic.utils.PageList;
import cn.yaboya.system.domain.Menu;
import cn.yaboya.system.mapper.MenuMapper;
import cn.yaboya.system.service.IMenuService;
import org.apache.poi.ss.formula.functions.T;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * <p>
 * 服务实现类
 * </p>
 *
 * @author yaboya
 * @since 2022-07-11
 */
@Service
public class MenuServiceImpl extends BaseServiceImpl<Menu> implements IMenuService {

    @Autowired
    private MenuMapper menuMapper;


    /**
    * @Title: loadAllTree
    * @Description: 无线极树
    * @Author: LT
    * @Version: 1.0
    * @Date:  2022/7/11 21:03
    * @Parameters: []
    * @Return java.util.List<cn.yaboya.system.domain.Menu>
    */
    @Override
    public List<Menu> loadAllTree() {


        // 1.一次性查询出全部数据
        List<Menu> menus = menuMapper.loadAll();

        // 2.创建一个集合对象，用于放最终排列完成的树数据
        List<Menu> menuTree = new ArrayList<Menu>();

        // 3.将集合数据换转为map，消除时间复杂度问题
        // Map<Long, Department> deptMap = deptTree.stream().collect(Collectors.toMap(Department::getId, item -> item));
        HashMap<Long, Menu> menuMap = new HashMap<>();
        for (Menu menu : menus) {
            menuMap.put(menu.getId(), menu);
        }

        for (Menu menuTmp : menus) {
            if (menuTmp.getParentId() == null) {
                menuTree.add(menuTmp);
            } else {
                // 从map集合中获取到父对象，将自己添加到父对象的children集合中
                Menu menu = menuMap.get(menuTmp.getParentId());
                if (menu != null) {
                    menu.getChildren().add(menuTmp);
                }
            }
        }

        return menuTree;
    }

}
